from openai import OpenAI
from vanna.chromadb import ChromaDB_VectorStore

from vanna_demo.config import my_config
from vanna_demo.deepseek_chat import DeepSeekChat


class MyVanna(ChromaDB_VectorStore, DeepSeekChat):
    def __init__(self, config, chroma_config=None):
        ChromaDB_VectorStore.__init__(self, config=chroma_config)
        DeepSeekChat.__init__(self, config=config)
      


vn = MyVanna(my_config, dict(path=my_config.chroma_path))
vn.connect_to_clickhouse(
    host=my_config.db_host,
    dbname=my_config.db_name,
    user=my_config.db_user,
    password=my_config.db_pass,
    port=my_config.db_port,
)